<?php
/**
 * @title 973. 最接近原点的 K 个点 - 最大堆
 * @author start2004
 */

// ini_set("memory_limit", "128M");

// class Solution {
//
//     function find() {
//
//     }
// }


class iMaxHeap extends SplMaxHeap{
    public function compare($arr1, $arr2)
    {
        if($arr1[0] == $arr2[0]){
            return 0;
        } elseif($arr1[0] > $arr2[0]) {
            return 1;
        } else {
            return -1;
        }
    }
}


class Solution {

    /**
     * @param Integer[][] $points
     * @param Integer $K
     * @return Integer[][]
     */
    function kClosest($points, $K) {
        /**
         * @since 2020-11-09 最大堆
         */
        $heap = new iMaxHeap();
        for($i=0; $i<$K; $i++){
            $d = $points[$i][0]*$points[$i][0] + $points[$i][1]*$points[$i][1];
            $heap->insert([$d, $points[$i]]);
        }

        /**
         * @since 2020-11-09 长度
         */
        $len = count($points);

        /**
         * @since 2020-11-09 保留最近的k个
         */
        for(; $i<$len; $i++){
            $d = $points[$i][0]*$points[$i][0] + $points[$i][1]*$points[$i][1];
            $data = $heap->top();

            /**
             * @since 2020-11-09 距离更短的点 入堆
             */
            if($d < $data[0]){
                $heap->extract();
                $heap->insert([$d, $points[$i]]);
            } else {}
        }

        /**
         * @since 2020-11-09 结果集
         */
        $result = [];
        for($i=0; $i<$K; $i++){
            $data = $heap->extract();
            $result[] = $data[1];
        }

        /**
         * @return
         */
        return $result;
    }
}



/**
 * @url http://127.0.0.1/leetcode/202011/2020.11.09.php
 */
$datas = [
    [[[1,3],[-2,2]], 1], // [-2,2]
    [[[3,3],[5,-1],[-2,4]], 2], // [3,3], [-2,4]
];

include_once dirname(__DIR__) . DIRECTORY_SEPARATOR ."xhprof.php";
$xhprof = new Xhprof();
foreach ($datas as $data){
    var_dump($data);

    $obj = new Solution();
    $result = $obj->kClosest(...$data);
    // $result = $obj->($xhprof->tree($data));
    // $result = $obj->($xhprof->listNode($data));
    // $result = $obj->find(...$data);
    var_dump($result);
    // if(count($result)<=20){
    //     var_dump($result);
    // } else {
    //     var_dump(count($result));
    // }
    echo str_repeat(PHP_EOL, 3);
    echo str_repeat("<br>", 3);
}

// foreach ($datas as $data){
//     $obj = new $data[0][0](...$data[1][0]);
//
//     for ($i=1; $i<count($data[0]); $i++){
//         $result = $obj->$data[0][$i](...$data[1][$i]);
//
//         echo $data[0][$i] ."(\"". implode(",", $data[1][$i]) ."\") ";
//             if($result === true){
//                 echo "True";
//             } elseif($result === false) {
//                 echo "False";
//             } elseif($result === null) {
//                 echo "Null";
//             } elseif(is_array($result)) {
//                 var_dump($result);
//             } else {
//                 echo $result;
//             }
//         echo PHP_EOL;
//     }
//
//     echo str_repeat(PHP_EOL, 3);
// }
$xhprof->end();
